from abc import ABC, abstractmethod
import torch.nn as nn

class BaseModelPlugin(ABC):
    """模型插件基类"""
    
    @abstractmethod
    def create_model(self, config):
        """创建模型实例"""
        pass
        
    @abstractmethod
    def get_model_params_count(self, model):
        """获取模型参数量"""
        pass